package com.zhuty.aiBioHelper.service;

import com.zhuty.aiBioHelper.enums.AiRoleType;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Component;

import java.util.HashMap;
import java.util.List;
import java.util.Map;

/**
 * AI服务工厂类
 * 负责根据角色类型创建和管理不同的AI服务实例
 * 
 * @author zhuty
 */
@Component
public class AiServiceFactory {
    
    private final Map<AiRoleType, BaseAiService> serviceMap = new HashMap<>();
    
    /**
     * 构造函数，自动注入所有AI服务实现
     * 
     * @param aiServices 所有AI服务实现的列表
     */
    @Autowired
    public AiServiceFactory(List<BaseAiService> aiServices) {
        // 将所有AI服务按角色类型注册到映射表中
        for (BaseAiService service : aiServices) {
            serviceMap.put(service.getSupportedRoleType(), service);
        }
    }
    
    /**
     * 根据角色类型获取对应的AI服务
     * 
     * @param roleType AI角色类型
     * @return 对应的AI服务实例
     * @throws IllegalArgumentException 如果找不到对应的服务
     */
    public BaseAiService getService(AiRoleType roleType) {
        BaseAiService service = serviceMap.get(roleType);
        if (service == null) {
            throw new IllegalArgumentException("未找到角色类型 " + roleType + " 对应的AI服务");
        }
        
        if (!service.isAvailable()) {
            throw new IllegalStateException("角色类型 " + roleType + " 对应的AI服务当前不可用");
        }
        
        return service;
    }
    
    /**
     * 根据角色代码获取对应的AI服务
     * 
     * @param roleCode 角色代码
     * @return 对应的AI服务实例
     */
    public BaseAiService getService(String roleCode) {
        AiRoleType roleType = AiRoleType.fromCode(roleCode);
        return getService(roleType);
    }
    
    /**
     * 获取所有可用的AI服务
     * 
     * @return 可用的AI服务映射表
     */
    public Map<AiRoleType, BaseAiService> getAllAvailableServices() {
        Map<AiRoleType, BaseAiService> availableServices = new HashMap<>();
        for (Map.Entry<AiRoleType, BaseAiService> entry : serviceMap.entrySet()) {
            if (entry.getValue().isAvailable()) {
                availableServices.put(entry.getKey(), entry.getValue());
            }
        }
        return availableServices;
    }
    
    /**
     * 检查指定角色类型的服务是否可用
     * 
     * @param roleType AI角色类型
     * @return 是否可用
     */
    public boolean isServiceAvailable(AiRoleType roleType) {
        BaseAiService service = serviceMap.get(roleType);
        return service != null && service.isAvailable();
    }
    
    /**
     * 检查指定角色代码的服务是否可用
     * 
     * @param roleCode 角色代码
     * @return 是否可用
     */
    public boolean isServiceAvailable(String roleCode) {
        try {
            AiRoleType roleType = AiRoleType.fromCode(roleCode);
            return isServiceAvailable(roleType);
        } catch (IllegalArgumentException e) {
            return false;
        }
    }
}